package com.htsc.mrp.utils;

import java.io.*;
import java.util.Enumeration;
import java.util.Properties;

public class PropertiesUtil {
    static final String GENERATOR_PROPERTIES_ADDRESS = "src/main/resources/generatorInit.properties";

    //根据key读取value
    static String readValue(String filePath, String key) {
        Properties props = new Properties();
        try {
            InputStream in = new BufferedInputStream(new FileInputStream(filePath));
            props.load(in);
            String value = props.getProperty(key);
            return value;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    //读取properties的全部信息
    static void readProperties(String filePath) {
        Properties props = new Properties();
        try {
            InputStream in = new BufferedInputStream(new FileInputStream(filePath));
            props.load(in);
            Enumeration en = props.propertyNames();
            while (en.hasMoreElements()) {
                String key = (String) en.nextElement();
                String Property = props.getProperty(key);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    //写入properties信息
    static void writeProperties(String filePath, String parameterName, String parameterValue) {
        Properties prop = new Properties();
        try {
            InputStream fis = new FileInputStream(filePath);
            //从输入流中读取属性列表（键和元素对）
            prop.load(fis);
            //调用 Hashtable 的方法 put。使用 getProperty 方法提供并行性。
            //强制要求为属性的键和值使用字符串。返回值是 Hashtable 调用 put 的结果。
            OutputStream fos = new FileOutputStream(filePath);
            prop.setProperty(parameterName, parameterValue);
            //以适合使用 load 方法加载到 Properties 表中的格式，
            //将此 Properties 表中的属性列表（键和元素对）写入输出流
            prop.store(fos, "Update '" + parameterName + "' value");
        } catch (IOException e) {
            System.err.println("Visit " + filePath + " for updating " + parameterName + " value error");
        }
    }

    //读取并修改properties文件
    public synchronized static void updateGeneratorInit(
            String targetProject,
            String targetResources,
            String modelTargetPackage,
            String mapperTargetPackage,
            String daoTargetPackage,
            String jdbcUrl,
            String jdbcUser,
            String jdbcPassword,
            String jdbcTableName
    ) {
        //target.project
        readValue(GENERATOR_PROPERTIES_ADDRESS, "target.project");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "target.project", targetProject);
        //target.resources
        readValue(GENERATOR_PROPERTIES_ADDRESS, "target.resources");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "target.resources", targetResources);
        //model.target.package
        readValue(GENERATOR_PROPERTIES_ADDRESS, "model.target.package");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "model.target.package", modelTargetPackage);
        //mapper.target.package
        readValue(GENERATOR_PROPERTIES_ADDRESS, "mapper.target.package");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "mapper.target.package", mapperTargetPackage);
        //dao.target.package
        readValue(GENERATOR_PROPERTIES_ADDRESS, "dao.target.package");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "dao.target.package", daoTargetPackage);
        //jdbc_url
        readValue(GENERATOR_PROPERTIES_ADDRESS, "jdbc_url");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "jdbc_url", jdbcUrl);
        //jdbc_user
        readValue(GENERATOR_PROPERTIES_ADDRESS, "jdbc_user");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "jdbc_user", jdbcUser);
        //jdbc_password
        readValue(GENERATOR_PROPERTIES_ADDRESS, "jdbc_password");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "jdbc_password", jdbcPassword);
        //jdbc_tableName
        readValue(GENERATOR_PROPERTIES_ADDRESS, "jdbc_tableName");
        writeProperties(GENERATOR_PROPERTIES_ADDRESS, "jdbc_tableName", jdbcTableName);

        readProperties("src/main/resources/generatorInit.properties");
    }
}